from sqlmodel import SQLModel

from entity.model.common_model import DataCollectionCommunity,  \
    BaseMixin, DataCollectionBuilding


# 小区表
class HeFeiDataCollectionCommunity(DataCollectionCommunity, table=True):
    __tablename__ = "data_collection_community_3401"


class HeFeiDataCollectionBuilding(DataCollectionBuilding, table=True):
    __tablename__ = "data_collection_building_3401"

# 房源管理-楼栋表
# class HeFeiCommunityBuildingRecord(BaseCommonModel, table=True):
#     __tablename__ = "crawler_hefei_community_building_record"
#     id: int = Field(
#         sa_column=Column(UnsignedBigInt, primary_key=True, nullable=False, comment="主键ID")
#     )
#     province: str = Field(
#         sa_column=Column(String(150), comment="省份")
#     )
#     area: str = Field(
#         sa_column=Column(String(150), comment="城市")
#     )
#     county: str = Field(
#         sa_column=Column(String(150), comment="区域")
#     )
#     province_id: int = Field(
#         sa_column=Column(UnsignedBigInt, comment="省份ID")
#     )
#     area_id: int = Field(
#         sa_column=Column(UnsignedBigInt, comment="城市ID")
#     )
#     county_id: int = Field(
#         sa_column=Column(UnsignedBigInt, comment="区域ID")
#     )
#     community_name: str = Field(
#         sa_column=Column(String(150), comment="小区名称")
#     )
#     community_id: str = Field(
#         sa_column=Column(String(100), comment="采集到的小区id")
#     )
#     community_href: str = Field(
#         sa_column=Column(String(255), comment="采集到的小区href")
#     )
#     address: str = Field(
#         sa_column=Column(String(255), comment="地址")
#     )
#     developer: str = Field(
#         sa_column=Column(String(120), comment="开发商")
#     )
#     building_id: str = Field(
#         sa_column=Column(String(100), comment="采集到楼栋id", unique=True)
#     )
#     building_href: str = Field(
#         sa_column=Column(String(255), comment="采集到的楼栋href,跳转到房间页面")
#     )
#     regulatory_banks: str = Field(
#         sa_column=Column(String(100), comment="监管银行")
#     )
#     regulatory_account: str = Field(
#         sa_column=Column(String(100), comment="监管账号")
#     )
#     certificate_no: str = Field(
#         sa_column=Column(String(100), comment="证书编号")
#     )
#     building_number: str = Field(
#         sa_column=Column(String(100), comment="楼栋号")
#     )
#     data_source: int = Field(
#         sa_column=Column(UnsignedInt, default=1, comment="数据来源:1-采集,0-购买")
#     )
